package com.wondersgroup.framework.sample.component.data;

import com.wondersgroup.framework.sample.component.tree.vo.CheckTreeVo;
import com.wondersgroup.framework.sample.component.tree.vo.FullTreeVo;
import com.wondersgroup.framework.sample.component.tree.vo.TreeVo;
import java.util.HashMap;
import java.util.List;
import javax.servlet.ServletContext;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;

public class TreeData
{
  private ServletContext context;

  public TreeData(HttpServletRequest request)
  {
    this.context = request.getSession().getServletContext();
  }

  public HashMap<CheckTreeVo, String> getTreeData()
  {
    return ((HashMap)this.context.getAttribute("treeList"));
  }

  public HashMap<TreeVo, String> getUnCheckTreeData() {
    return ((HashMap)this.context.getAttribute("uncheckTreeList"));
  }

  public List<FullTreeVo> getFullTreeData()
  {
    return ((List)this.context.getAttribute("fullTreeList"));
  }
}